public class StrIndexOf {
    /*
    * ----------------需要时常复习-----------------
    * 找出字符串中第一个匹配项的下标
    * 给你两个字符串 haystack 和 needle ，
    * 请你在 haystack 字符串中找出 needle 字符串的第一个匹配项的下标（下标从 0 开始）。
    * 如果 needle 不是 haystack 的一部分，则返回  -1 。
    *
    * 1 <= haystack.length, needle.length <= 10(4)
    * haystack 和 needle 仅由小写英文字符组成
    * */
    public static void main(String[] args){
        StrIndexOf strIndexOf = new StrIndexOf();
        String haystack = "sdsdsdsdsad";
        String needle = "sad";
        System.out.println(strIndexOf.solution("sdsdsdsdsad", "sad"));
        System.out.println(haystack.indexOf(needle));
        int[] a = new int[5];
    }

    // 我的想法：遍历一遍haystack， 每当遇到needle开头就看看是不是子串
    public int mySolution(String haystack, String needle){
        // 注意，题目没说needle长度一定小于haystack，得判断！
        if(needle.length() > haystack.length())
            return -1;
        for (int i = 0; i < haystack.length() ; i++) {
            if (haystack.charAt(i) == needle.charAt(0)){
                int flag = i++;
                int j = 1;
                // 这里的条件一定是j < needle.length() && i < haystack.length(), 否则会越界
                for ( ; j < needle.length() && i < haystack.length(); j++) {
                    if (haystack.charAt(i) == needle.charAt(j)) {
                        i++;
                        continue;
                    }
                    else
                        break;
                }
                if (j == needle.length())
                    return flag;
                else {
                    i = flag;
                }
            }
        }
        return -1;
    }


    // kmp算法，建立前缀表
    public int solution(String haystack, String needle){
        if(needle.length() > haystack.length())
            return -1;
        // 建立前缀表
        int[] next = new int[needle.length()];
        getNext(next, needle);
        int j = 0;
        for (int i = 0; i < haystack.length(); i++) {
            // 不匹配，j按照前缀表回退
            while (j > 0 && needle.charAt(j) != haystack.charAt(i))
                j = next[j-1];
            // 匹配，向后走
            if(needle.charAt(j) == haystack.charAt(i))
                j++;
            // 匹配完了，返回
            if(j == needle.length())
                return i - j + 1;
        }
        return -1;
    }

    // 获得前缀表
    // 1.初始化值
    // 2.前后缀不匹配的情况
    // 3.前后缀匹配的情况
    // 4.更新next的值
    public void getNext(int[] next, String needle){
        // 1.初始化
        int j = 0;
        next[0] = 0;
        for (int i = 1; i < needle.length(); i++) {
            // 2.前后缀不匹配的情况 循环跳回直，至匹配或回到开头
            while (j > 0 && needle.charAt(j) != needle.charAt(i))
                j = next[j-1];
            // 3.前后缀匹配的情况, j++,获得当前的前缀长度
            if (needle.charAt(j) == needle.charAt(i))
                j++;
            // 4.更新next的值
            next[i] = j;
        }
    }
}
